//
// Created by 紫霄 on 2020/4/24.
//

#ifndef LEETCODE_56_H
#define LEETCODE_56_H
#include <algorithm>
#include <complex>
using namespace std;
struct TreeNode {
  int val;
  TreeNode *left;
  TreeNode *right;
  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

// class Solution {
// public:
//  bool isBalanced(TreeNode *root) {
//    if (!root) {
//      return true;
//    }
//    int leftHeight = getHeight(root->left);
//    int rightHeight = getHeight(root->right);
//    int heightDiff = leftHeight - rightHeight;
//    if (heightDiff < 2 && -2 < heightDiff && isBalanced(root->left) &&
//        isBalanced(root->right)) {
//      return true;
//    }
//  }
//  int getHeight(TreeNode *root) {
//    if (root) {
//      return max(getHeight(root->left), getHeight(root->right)) + 1;
//    } else {
//      return 0;
//    }
//  }
//  int max(int a, int b) { return a > b ? a : b; }
//};

class Solution {
public:
  bool isBalanced(TreeNode *root) {
    int depth = 0;
    return isBalanced(root, depth);
  }
  bool isBalanced(TreeNode *root, int &depth) {
    if (!root) {
      return true;
    }
    int rDepth = 0;
    int lDepth = 0;
    if (isBalanced(root->left, lDepth) && isBalanced(root->right, rDepth)) {
      if (abs(lDepth - rDepth) < 2) {
        depth = 1 + max(lDepth, rDepth);
        return true;
      }
    }
    return false;
  }
};

#endif // LEETCODE_56_H
